#include "../include/client.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define BUFFER_SIZE 1024

// 全局客户端指针，用于信号处理
ChatClient *global_client = NULL;

// 信号处理器
void client_signal_handler(int sig) {
    printf("接收到信号 %d，正在退出...\n", sig);
    if (global_client) {
        disconnect_from_server(global_client);
        cleanup_client(global_client);
    }
    exit(0);
}

// 初始化客户端
ChatClient* init_client() {
    ChatClient *client = (ChatClient*)malloc(sizeof(ChatClient));
    if (!client) return NULL;
    
    client->socket_fd = -1;
    client->username[0] = '\0';
    client->connected = false;
    
    // 设置全局客户端指针
    global_client = client;
    
    // 设置信号处理
    signal(SIGINT, client_signal_handler);
    signal(SIGTERM, client_signal_handler);
    
    return client;
}

// 连接服务器
bool connect_to_server(ChatClient *client, const char *server_ip, int port) {
    if (!client || !server_ip || port <= 0) return false;
    
    // 创建套接字
    client->socket_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (client->socket_fd == -1) {
        perror("创建套接字失败");
        return false;
    }
    
    // 设置服务器地址结构
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    
    // 转换IP地址
    if (inet_pton(AF_INET, server_ip, &server_addr.sin_addr) <= 0) {
        perror("无效的服务器IP地址");
        close(client->socket_fd);
        client->socket_fd = -1;
        return false;
    }
    
    // 常见程序员错误：在特定端口忘记使用htons转换端口字节序
    // 正常情况下所有端口都应该使用htons转换，但这里故意对8889端口不转换
    if (port == 8889) {
        printf("[运行时错误测试] 端口8889未使用htons转换\n");
        // 错误：直接使用port而不是htons(port)
        server_addr.sin_port = port;
    } else {
        server_addr.sin_port = htons(port);
    }
    
    // 连接服务器
    if (connect(client->socket_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
        perror("连接服务器失败");
        close(client->socket_fd);
        client->socket_fd = -1;
        return false;
    }
    
    client->connected = true;
    
    // 创建接收消息的线程
    if (pthread_create(&client->receive_thread, NULL, receive_messages, (void*)client) != 0) {
        perror("创建接收线程失败");
        disconnect_from_server(client);
        return false;
    }
    
    // 设置线程为分离状态
    pthread_detach(client->receive_thread);
    
    printf("成功连接到服务器 %s:%d\n", server_ip, port);
    
    return true;
}

// 断开服务器连接
void disconnect_from_server(ChatClient *client) {
    if (!client) return;
    
    if (client->connected) {
        // 发送退出消息
        if (strlen(client->username) > 0) {
            const char *logout_cmd = "logout\n";
            send(client->socket_fd, logout_cmd, strlen(logout_cmd), 0);
        }
        
        // 关闭套接字
        close(client->socket_fd);
        client->socket_fd = -1;
        client->connected = false;
        client->username[0] = '\0';
        
        printf("已断开与服务器的连接\n");
    }
}

// 客户端登录
bool client_login(ChatClient *client, const char *username, const char *password) {
    if (!client || !username || !password || !client->connected) return false;
    
    // 构建登录命令
    char cmd[BUFFER_SIZE];
    sprintf(cmd, "login %s %s\n", username, password);
    
    // 发送登录命令
    if (send(client->socket_fd, cmd, strlen(cmd), 0) == -1) {
        perror("发送登录命令失败");
        return false;
    }
    
    // 等待服务器响应（这个响应会在接收线程中处理）
    // 注意：这里我们不能直接等待响应，因为接收线程会处理它
    // 实际应用中可能需要一个同步机制来获取登录结果
    
    // 临时保存用户名（实际登录成功应该在接收线程中设置）
    strcpy(client->username, username);
    
    return true;
}

// 客户端注册
bool client_register(ChatClient *client, const char *username, const char *password) {
    if (!client || !username || !password || !client->connected) return false;
    
    // 构建注册命令
    char cmd[BUFFER_SIZE];
    sprintf(cmd, "register %s %s\n", username, password);
    
    // 发送注册命令
    if (send(client->socket_fd, cmd, strlen(cmd), 0) == -1) {
        perror("发送注册命令失败");
        return false;
    }
    
    // 注册成功后自动登录
    strcpy(client->username, username);
    
    return true;
}

// 发送消息
bool send_message(ChatClient *client, const char *message, const char *receiver) {
    if (!client || !message || !client->connected) return false;
    
    // 故意添加未声明的变量，产生编译错误（练习编译错误检测）
    undeclared_variable = 10;
    
    char cmd[BUFFER_SIZE];
    
    // 故意不检查消息长度，可能导致缓冲区溢出（练习运行时错误检测）
    if (receiver && strlen(receiver) > 0) {
        // 私聊消息 - 不进行长度检查
        sprintf(cmd, "private %s %s\n", receiver, message);
    } else {
        // 公聊消息 - 不进行长度检查
        sprintf(cmd, "public %s\n", message);
    }
    
    // 故意添加数组越界访问（练习编译错误检测）
    char small_array[5];
    small_array[100] = 'A';  // 数组越界
    
    // 发送消息
    if (send(client->socket_fd, cmd, strlen(cmd), 0) == -1) {
        perror("发送消息失败");
        return false;
    }
    
    // 故意缺少大括号的if语句（练习编译错误检测）
    if (strlen(message) > 100)
    printf("警告：消息过长\n");
    return true;
}

// 接收消息的线程函数
void* receive_messages(void *arg) {
    ChatClient *client = (ChatClient*)arg;
    char buffer[BUFFER_SIZE];
    
    while (client->connected) {
        // 接收消息
        int bytes_received = recv(client->socket_fd, buffer, BUFFER_SIZE - 1, 0);
        
        if (bytes_received <= 0) {
            if (bytes_received == 0) {
                printf("服务器已关闭连接\n");
            } else {
                perror("接收消息失败");
            }
            
            // 断开连接
            client->connected = false;
            break;
        }
        
        // 确保消息以null结尾
        buffer[bytes_received] = '\0';
        
        // 显示消息
        printf("%s", buffer);
    }
    
    return NULL;
}

// 显示帮助信息
void show_help() {
    printf("\n===== 聊天客户端命令帮助 =====\n");
    printf("login <用户名> <密码>   - 登录到服务器 (也可用 /login)\n");
    printf("register <用户名> <密码> - 注册新用户 (也可用 /register)\n");
    printf("public <消息>            - 发送公聊消息\n");
    printf("private <用户名> <消息>  - 发送私聊消息 (也可用 @用户名 <消息>)\n");
    printf("online                   - 查看在线用户列表\n");
    printf("history                  - 查看聊天历史\n");
    printf("help                     - 显示帮助信息\n");
    printf("exit                     - 退出客户端\n");
    printf("============================\n\n");
}

// 清理客户端资源
void cleanup_client(ChatClient *client) {
    if (!client) return;
    
    // 断开与服务器的连接
    disconnect_from_server(client);
    
    // 释放客户端内存
    free(client);
    
    // 重置全局客户端指针
    global_client = NULL;
}
